Implement WAILA plugin. Some cleanup.
WAILA has the best API...