Removed __main__ part and pep8