Basic algorithm done

Need to add in automated cross validation