Integrate command-line prompts into Python script