Updated script for testing