Remove import * and PEP8 formatting cleanup
Add command line arguments