Add sourcesets to the extension to allow configuration.
Fix tests for new functionality.
Merge pull request #85 from SuperJugy/test

Fixes #32, Closes #83.