adds extension method/ refactor