The only UI edit I would suggest would be the notification area of the activation like "Belts direction inverted" be moved or made less intrusive. On some of the later, larger, levels it would cover up some switches/buttons that sometimes needed to be precisely interacted with (turn off a belt then rapidly navigate stopped belts.