Tidy the code...