Implement a Lazy Array | Databricks Coding Question