Python Function Return Variable Number Definition Math