Add headers support to flash. Fix tests for recent refactored yaml keys.