Misc: tidy up code