added screenshots, update code