Adds command-line argument support