Small changes, bugfix

Possible error in Grad-B drift term