Add stubs for the class and test modules