Add shape class