removing unneeded method and adding class that I forgot to add.