Thanks TDD! My code was much more general than I thought