break function in if statement python