Debug new feature: use idx which contains id and index of molecule info