quick code tidy