make code tidy