Code from Day 5
, which you can use for Q1.