Forgot to add new class file