Stylistic changes to fit more into theano.
pep8/stylistic changes