adding zaxis and some pep8