Add support for source map generation