I've changed this... I've also made some modifications for the other particles, but that's currently in another document where I'm trying to clean up and reduce the code (modtest.py, it's not done yet).