Fixed some plotting and added a lot of comments

* Changed the colormap on contour plot
* Modified plot streamline
* Added a lot of comments so `doc` and `help` functions work better...
still a lot to do