New time gen

Finalize and address Alex's comments, trial 2.
I did not fully unpack the end of the line :) this code works because train times is a dict already, got it