added heatplot and R_squared measure of estimates