Updating README
Some code formatting, clean up, and commenting