Simplify flash mock
Implement flash writing command