Some minor changes/clarifications. Also added grid line to plot function