Cleaned up brownian motion to reflect changes to jittermodel code