Exercise 4 from Day 4 of Hackbright, fixing functions for a testing file