Add test for updating survey answers