Adding screenshot correction