Moved decorate method in to correct class