added pictures to code code listing