Change to one main controller