tidies up code