remove extra comment
clean up trainer code