work on python script for calculating particle flux