Use decorator for code reuse