expanded script fixed tests for demo example