minor tweaks

fixed typo
slight optimalizations
added new tag {% train %}