Add some new features