Adding test classes