DEMO: Update initialization in LSSM demo