Updated example to also vary the shape