Allow passing source as array on edit

Closes #6