pep8 formatting and added main