refactor
Scripts for demo