Tidy up some module