Hmm. I'm not entirely convinced it would be clearer to add the code directly to `__init__`. The way it is now keeps all the key logic grouped together a bit more closely. Any other opinions on this? It would be an easy change of course.