"Added stuff for Challenge 2"