Small change to main to allow dynamic input