Tidies up the code.