Update UnitsController.php