Add two test classes