Added more dialogue functionality