Cosmit some pep8
fix multi-class test