The code is now working.

Todo list:
- add neutral particles
- get input from file or arguments