Remove call to non-existing method get_batch_axis(). Test included
fix pep8 error
Enable other mlp tests
changed john's fix.