Updated polymorphism exercise to use asserts