Added pass to trainer update method