Add initial controller tests