Add sourcesets to the extension to allow configuration.

Should fix #32. #83 should not be needed anymore.