screenshot + code improvments

screenshot add to the project and a few code tweaks