Edits to molecule viewer