Add plotting function for simple regression context