Updated remove method to handle removing first link, added tests