Added methods to controller test class