elseif shape.mode == 'polygon' then

  elseif shape.mode == 'polygon' or shape.mode == 'rectangle' or shape.mode == 'square' then