Visualizer: Fixed add_unit and init_units