Code revision

Re-add comments where necessary! :P 
Hope this helps! :palm_tree: