Modified main code to reflect pep8 standard