DEMO: Improve initialization of switching LSSM in demo