Improved an error message
Added initial iteration of plot_experiments
Removed default color
Added disable_jit() functionality.
Some minor docstrings clean ups and removed unused imports.
Removed extra code.
Removed unused functions