Add test for friend remove button