how to return a dictionary in python function