Update the generated controller class

in order to use the setItemDisplayed method