Add test for the reply feature