Tidy code 3