Visual improvement in demo