Forgot to add new class files