Add test for case classes