Fixed bugs in Transformer, moving on to solve