added classes to images coming in