adding better k-fold based on issue #57