Add test cases for friends and a TODO