added screenshots and couple of new tests