Added another case to remove method