Add units_number_to_keep to specifiy which units to keep at any cost