Dive Computer Function Definition Python