Implement exercise 1 of week 2