Fix some things
Merge pull request #4 from achyutreddy24/master

Fix some things