Added some comments and cleaned up code
Added some comments and cleaned up code