Add test for inner classes