Small update in demo app