Added some "WAILA" integration