tidy up source code

made it prettier.