Add screenshots and make small layout improvements