Updated script to include new parameters