python list remove and return