revised code could help improve efficiency of fusion experiments