Updated example script to include command line arguments