Fix image grid layout