Forgot to add the new class :)