Added follow and unfollow logic, and passed the unit test