Remove some commented-out code
Update Zernike demo notebook to match changed API