made the code a little prettier