Very small update to script