change controller code