Add tests for Molecule class