trying to fix the image layout issue