Drop the Warm-Up and Cool-Down — medium Python Fundamentals problem | Incognition