add requirements file
main on top (it is the first function to read, I don't want to scroll) and arguments in a separate function (separation of concerns)
removed unused import statements
removed useless comment
defining __init__ on top is a common best practice
space around operators and after commas (PEP8)