Update with new features