Adds style for error section
Modifies behaviour for error, working correctly for one image at time