Imp use of .size instead of np.prod(shape)
Fix typo + pep8 + pyflakes