Include a unit test for the new controller