training model works now