Changed one class to reflect some unit logistics.