Kriminalvården, Utbildningsenheten