Add dialog support