adding screenshot and remaining code