pep8 math operators
use np.rollaxis
reference formatting