Fixed demo script