Lecture 6 changes

Used polymorphism on the view controllers.