Fix molecule.py