added class for predicitons and output