Cleaning up stuff
Cleaning up a few more things. Added a few lines of documentation