Adding some stuff to work with the image loading