Tidy up UIModule