Reorganize molecular dynamics tasks