slight update. still need to implement cross validation, but everything else is in report