Updated controller example