Updated script for demoing