screen shots added
Changes to code logic