adding classes to partial_fit