modified run_lj.py to make it compatible with new parallelisation method