Automate screenshot taking

This has IMO low priority but it would be great if we could automate the screenshot generation e.g. using [Selenium](https://www.seleniumhq.org/). It would make it more reproducible and speed up things on the long run.