move cross_validation methods to its own function, add plotting wrapper in problem_4.m