Add new logic