Changed controller class on UI