* add flash module * remove todo * run tests Co-authored-by: Harshil Patel <drpatelh@users.noreply.github.com>