Add support for source-map generation