Add test for friend removal