Minor tweaks for demo.
Updating to support multiple balls.