A bit of cleanup
Added getRMSE method