Fixes to mystats demo script