Add some controller tests