Added WAILA support.