Update puck_simple_example.py