Slight restructuring
Added k fold properly