Updating script for figures