Cleaned up code

- 4 space indent
- more comments
- minor refactor
Added README