Added logic to the decorator that marks fixed errors.
Refactored code.