Adding class to work with uploaded images