Cleaning up the code

I fixed the formatting mostly