Switched to command-line input and updated readme