Fixing so remove function can handle only string values