Tidy the code
Tidy the code