Added function to plot decision rules. Import it by default.