Imafe code tidy