fix stupid code & add screenshots