Added controller test class