Improve montypuzzle example