Do you think we should get some sort of slicing into MDTraj?  Are you OK with `atom(i)`?