Adding controller tests