Quick code tidy