Train a small GPT to solve mazes