Added options to pass in key word arguments to the update_sigma_fn in Brownian Dynamics simulation

Bd improvement 2

In response to @cpgoodri 's comment. Rolling back to the original definition to ensure backward compatibility, and updated the documentation to clarify the definition of gamma. Also, added the option for user to pass in gamma in the apply_fn similarly to how temperature can be passed in.