Feature/set source

Fixes #1