nicer plots, and some comments in code