Updated demo script for GUI