Updated example to work with the new impl.
Fixed retake button not working.