Minor edit
Reduced the bath size to control memory used