Changes to ViewController to complete Challenge 1