cleaned up the code

...kinda

by cleaned up I mean I made it look pretty