Clean up molecular dynamics benchmark