Add work display function