Add arguments to control number of diffusion steps and number of recycles

[marcuscollins]

These should be piped all the way from run_grid_search.py to the individual Trials

[k-chrispens]

@marcuscollins is this done with #205 ?

[marcuscollins]

Yes, closing with #205