[python] Python - Passing a function into another function