Fixed some code so it looks nicer, because I can.