Updated part of code to follow pep8 standard