Update demo to use the 0.5.1 release

Adding models now requires a unique id.