homework 3 . Add interface for use polymorphism