Passing Function As An Argument In Python