Add test case for follow function