Handling failures in YARN