Update unit buttons