Modified the  (1+1)-d code. Also the notebook