Implement in-memory DB querying | OpenAI Coding Question