added screenshots
updated code