Added command line argument for staging