Challenge 4 for the Udemy iOS course