Add some function to color the box and draw text