Adding controller test