Fixes second instance of 20% bug when splitting data set into optimization and testing sets.