added code for the 2 screenshots