Fixed classification test, added larger dataset