(1) Nothing contained in § 40-35-302, § 40-35-306, § 40-35-307 or § 40-35-314 shall prevent the official having responsibility for the custody and safekeeping of the defendant, as an alternative to the judicial remedy provided in those sections, from duly promulgating and enforcing reasonable disciplinary rules and procedures that provide for appropriate punishments for violations of any condition or rule of the institution in which the defendant is incarcerated, such as increasing the amount of time the defendant must serve in confinement or by changing the conditions of the defendant's confinement, or both. Any increase in the amount of time to be served shall not exceed the full sentence originally imposed by the court.
(2) Officials having responsibility for the custody and safekeeping of defendants may promulgate and enforce reasonable disciplinary rules and procedures requiring all able-bodied inmates to participate in work programs. The rules and procedures may provide appropriate punishments for inmates who refuse to work, including, but not limited to, increasing the amount of time the defendant must serve in confinement or changing the conditions of the defendant's confinement, or both. Any increase in the amount of time a defendant must serve for refusing to participate in a work program shall not exceed the sentence originally imposed by the court.