Moved prediction methods into model class.