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.