Implement in-memory DB querying | OpenAI Interview Question