add cutomized training