Change memorise to diff, add turn on function and extra test