Discipline — the KJV “chastening” — is addressed in Proverbs and in the Epistle to the Hebrews.
In Proverbs 3:11-12:
11 My son, despise not the chastening of the LORD; neither be weary of his correction: 12 For whom the LORD loveth he correcteth; even as a father the son in whom he delighteth.
In Hebrews 12:6:
6 For whom the Lord loveth he chasteneth, and scourgeth every son whom he receiveth.
In Proverbs 13:24:
24 He that spareth his rod hateth his son: but he that loveth him chasteneth him betimes.