Add lens drawing function