Add Shape trait