Add flake

Closes #4.