update for new features