Tidy the code