Minor changes on work machine