Worked on fixing regression, trimming nans
Working on loading validation for all runs