More updates to demo script