Use controller in demo