add function to paint head