corrected the 5 fold validation