Updated example for new features