Cleaned up code
Changed up code format
Merge pull request #8 from suketk/master

Yeah son!