Cleaned up code
Merge pull request #2 from divyeshbalaraaaa/master

Cleaned up code