Add omnia and mdtraj