added screenshot and code fixes