Remove unneeded imports
Remove incorrect variable setting
Removing header hack
Forgot to save part of the notebook
Update version