Add legend and remove AST namespace