set up for optimization with more generations, not sure if working yet