Added controller test