Fixed bugs and compared against existing simulation. Added verification notebook.