Updating to ensure pep8 compliaance