Some updates to the physics demo