Tiny refactor
Add screenshots