Added new class that was forgotten