Add better dialog system