Is the main function of prison/jail to rehabilitate people?

I mean correctional facility means to "correct" people--yes of course there is the punishment aspect, however we know most people will be out so let's try to make people better and rehabilitate. The USA has the most punitive justice system in the industrialized world with the most prisoners in the world and has the highest violent crime rate in the industrialized world--go figure.