was able to implement the class inheritance bonus problem