Adding shape trait