adapted pretrained.py to handle neutrals