added pictures in code section