Follow the Arrow — medium Python Fundamentals problem | Incognition