Convex Functions In Python