Some cleanup and minor rearanging the code.
New notebook with research on analytical solution