Added feature for testing screenshots