Added functions for prompt and command line arguments