added option to retrain proportions