Add polyco class (in predictor.py)