Corrected class name.

Trying to reuse some code from the other assignements and forgot to change the class name