Implement follow graph with snapshots | OpenAI Interview Question