Exercise 9 - updated to ask user which months cohort