Fixed setup to use commandline arguments
Fixed an issue.
Merge pull request #4 from miheerdew/FixSetup

Fix setup