Add controller test
Add test