Optimize Functions In Python