Code Challenge 4 additions