Implement lazy array and KV store | Databricks Interview Question