cleanup code
Merge pull request #9 from vinitkumar/develop

cleanup code