Forgot to add new class.