Add shape addition code