Small changes to molecule.py