Mallet Clean code #5

*fixed mallet trainer