; (B) who is vested with authority to bear arms and make arrests; and (C) whose primary responsibility is the prevention and detection of crime or the enforcement of the penal, criminal, traffic, or highway laws, including any supervisory or command personnel whose duties include the supervision, training, guidance, or management of any such person. (31) The term correctional officer means a person— (A) who is appointed or employed by the Federal Government or by a State or political subdivision thereof, and whose primary responsibility is the supervision, protection, care, custody, and control, or investigation, of inmates of a correctional institution; or (B) who is employed by any private entity that has contracted with the Federal Government or a State or political subdivision thereof, and who has such a primary responsibility. . (d) Effective date The amendments made by this section shall apply to conduct engaged in after the 6-month period that begins with the date of enactment of this Act.