Added controller unit test