adding more menu choices, changing the workflow, adding some error handling