Add test for harder problem