Made my code a little neater.