How To Pass Function As Argument Into A Function For Python