updated train yourself and added function.: