Error with model.batch_removal(): AttributeError: 'NoneType' object has no attribute 'sqrt'

[macrophage666]

Thanks for providing such a great tool! I found this error when I run the model.batch_removal() function

[LouiseDck]

I've ran into the same issue and found a fix that worked for me. See #104