Add source map generation