changed line 16 from "Train..." to "Decode..."