working, curr using the best model