Added working controller to display layout