Reworked the old notebook hack

now also cells work correctly