Add flake support