Update Km_Controller.php