Tidy code :)