Python Function To Get Input