Refactor demo script

And delete images before each run