Add a function so that I can also save the molecule width.