Tidy the code up