Added better CI

@rcrehuet  I also fixed some issues with the `ensure_type` import from mdtraj not working with the latest version of mdtraj.