Updated demo code for MAZE+.