updates for the new feature