Format imports according to pep-8. Fix parsing arguments by using argparse.
Add few more comments to the code.
Merge pull request #3 from xa4a/master

Format imports according to pep-8. Fix parsing arguments by using argparse. Add couple comments to the code.