If Function Returns True Python