Remove the currying function

Replace it with functools.partial