added testing framework for pre-trained models
messed with ae stuff