Adding test class