Wrote controller tests