Python Function Return Variable Number Definition