Changed taking images from album class now