Added Comments

- Added comments for the rest of the code
- Very minor aesthetic changes around the code
- Updated TODO list