Still trying to get polymorphism to work properly