Add screenshot functionality. Step1