Forgot to add the new classes