Implement follow graph with snapshots | OpenAI Coding Question