How To Take User Input In Dictionary In Python Functions