Update code (refine drawing function)