Add functionality for command line arguments