minor changes for final demo