cleaned up the code
Merge pull request #9 from Suren6212/clean-up

cleaned up the code