Update Diamond_Polymer.py