Single Variable Function Reduce Python Function Return