Updated example to use the shape module