Can now take a command-line argument