Python Average Function Code