What is the primary function of a lens in optical devices?