TODO: Split for 3 fold validation