Adding unit test for controller