Dive Computer Function Definition Python Return