Add new class for testing unit test