Added model for holding training data