The correctional system should be primarily rehabilitative, which is a part of the name "correctional". The criminals will eventually leave the premises of the facility when they're done with their time. If you only punish them, they'll have problems coming back to the society and might repeat their crimes. Rehabilitative facilities would help integrate them and correct their behavior so as to prevent them from leading a life of crime, which is not something that simple punishment would do.
