Make the code a little prettier