Extracted task class into the lib/ folder so real tests can be written