Move duplicate code to a reusable function