updated function, first GBM model