Updated to take command line arguments or input