Changed training input, added solution. Done.