changes to the main method
Merge pull request #1 from swaprad/master

changes to the main method