move custom attribute into seperate class