Save the Grid — medium Python for AI problem | Incognition