Python Function Return A Value Based