Fix and clean up initial example.