Renamed a class, needed to edit here also