The trainer script actually works now