Add the functionality to read input from the command line