Could you please change it to follow this user's[0] code instead?

[0] https://news.ycombinator.com/item?id=50190859