diff --git a/README.md b/README.md
index dca7210..ce88450 100644
--- a/README.md
+++ b/README.md
@@ -31,13 +31,13 @@ IE Mobile, UC Browser for Android.
 
 Further details can be found here:
 
-* [https://web.dev/samesite-cookies-explained](SameSite cookies explained)
+* [SameSite cookies explained](https://web.dev/samesite-cookies-explained)
 * [CSRF is (really) dead](https://scotthelme.co.uk/csrf-is-really-dead/)
 * [PHP setcookie “SameSite=Strict”?](https://stackoverflow.com/questions/39750906/php-setcookie-samesite-strict)
 * [How to Set a cookie attribute Samesite value in PHP ?](https://www.tutorialshore.com/how-to-set-a-cookie-attribute-samesite-value-in-php/)
 * [Can I use SameSite?](https://caniuse.com/#feat=same-site-cookie-attribute)
 
-## Usage
+## Slim 4 integration
 
 Slim 4 uses a LIFO (last in, first out) middleware stack,
 so we have to add the middleware in reverse order:
@@ -45,7 +45,8 @@ so we have to add the middleware in reverse order:
 ```php
 <?php
 
-use Selective\SameSiteCookie\SameSiteCookieMiddlware;
+use Selective\SameSiteCookie\SameSiteCookieConfiguration;
+use Selective\SameSiteCookie\SameSiteCookieMiddleware;
 use Selective\SameSiteCookie\SameSiteSessionMiddleware;
 use Slim\Factory\AppFactory;
 
@@ -53,11 +54,13 @@ $app = AppFactory::create();
 
 // ...
 
+$configuration = new SameSiteCookieConfiguration();
+
 // Register the samesite cookie middleware
-$app->add(new SameSiteCookieMiddlware(true));
+$app->add(new SameSiteCookieMiddleware($configuration));
 
 // Start the native PHP session handler and fetch the session attributes
-$app->add(new SameSiteSessionMiddleware('Lax', true, true));
+$app->add(new SameSiteSessionMiddleware($configuration));
 
 // ...
 
diff --git a/composer.json b/composer.json
index f6a8a79..3312c1c 100644
--- a/composer.json
+++ b/composer.json
@@ -17,9 +17,11 @@
         "psr/http-server-middleware": "^1.0"
     },
     "require-dev": {
+        "nyholm/psr7": "^1.1",
         "overtrue/phplint": "^1.1",
-        "phpunit/phpunit": "^7",
         "phpstan/phpstan-shim": "^0.11",
+        "phpunit/phpunit": "^7",
+        "relay/relay": "^2.0",
         "squizlabs/php_codesniffer": "^3.4"
     },
     "scripts": {
diff --git a/phpunit.xml b/phpunit.xml
index 1babff2..e3d5d55 100644
--- a/phpunit.xml
+++ b/phpunit.xml
@@ -1,5 +1,5 @@
 <?xml version="1.0" encoding="UTF-8"?>
-<phpunit bootstrap="vendor/autoload.php"
+<phpunit bootstrap="tests/boostrap.php"
          colors="true"
          backupGlobals="false"
          backupStaticAttributes="false"
diff --git a/src/SameSiteCookieConfiguration.php b/src/SameSiteCookieConfiguration.php
new file mode 100644
index 0000000..53145a6
--- /dev/null
+++ b/src/SameSiteCookieConfiguration.php
@@ -0,0 +1,29 @@
+<?php
+
+namespace Selective\SameSiteCookie;
+
+/**
+ * SameSite Cookie Configuration.
+ */
+final class SameSiteCookieConfiguration
+{
+    /**
+     * @var bool Start the session
+     */
+    public $startSession = true;
+
+    /**
+     * @var string Send cookie only via a href link. Values: 'Lax' or 'Strict'.
+     */
+    public $sameSite = 'Lax';
+
+    /**
+     * @var bool Prevents cookies from being read by scripts. Should be enabled.
+     */
+    public $httpOnly = true;
+
+    /**
+     * @var bool Provide cookies only via ssl. Should be enabled in production.
+     */
+    public $secure = true;
+}
diff --git a/src/SameSiteCookieMiddlware.php b/src/SameSiteCookieMiddleware.php
similarity index 57%
rename from src/SameSiteCookieMiddlware.php
rename to src/SameSiteCookieMiddleware.php
index ad6deae..e7da580 100644
--- a/src/SameSiteCookieMiddlware.php
+++ b/src/SameSiteCookieMiddleware.php
@@ -10,7 +10,7 @@
 /**
  * SameSite Cookie Middleware.
  */
-final class SameSiteCookieMiddlware implements MiddlewareInterface
+final class SameSiteCookieMiddleware implements MiddlewareInterface
 {
     /**
      * @var string
@@ -30,15 +30,13 @@ final class SameSiteCookieMiddlware implements MiddlewareInterface
     /**
      * The constructor.
      *
-     * @param string $sameSite Send cookie only via a href link. Values: 'Lax' or 'Strict'.
-     * @param bool $httpOnly Prevents cookies from being read by scripts. Should be enabled.
-     * @param bool $secure Provide cookies only via ssl. Should be enabled in production.
+     * @param SameSiteCookieConfiguration $configuration The configuration
      */
-    public function __construct(string $sameSite = 'Lax', bool $httpOnly = true, bool $secure = false)
+    public function __construct(SameSiteCookieConfiguration $configuration)
     {
-        $this->sameSite = $sameSite;
-        $this->httpOnly = $httpOnly;
-        $this->secure = $secure;
+        $this->sameSite = $configuration->sameSite;
+        $this->httpOnly = $configuration->httpOnly;
+        $this->secure = $configuration->secure;
     }
 
     /**
@@ -57,20 +55,6 @@ public function process(ServerRequestInterface $request, RequestHandlerInterface
         $sessionName = $request->getAttribute('session_name');
         $params = $request->getAttribute('session_cookie_params');
 
-        if (version_compare(PHP_VERSION, '7.3.0') >= 0) {
-            // Remove invalid key
-            unset($params['lifetime']);
-
-            $params['samesite'] = $this->sameSite;
-            $params['httponly'] = $this->httpOnly;
-            $params['secure'] = $this->secure;
-
-            setcookie($sessionName, $sessionId, $params);
-
-            return $response;
-        }
-
-        // For older PHP versions
         $cookieValues = [
             sprintf('%s=%s;', $sessionName, $sessionId),
             sprintf('path=%s;', $params['path']),
@@ -88,8 +72,7 @@ public function process(ServerRequestInterface $request, RequestHandlerInterface
             $cookieValues[] = sprintf('SameSite=%s;', $this->sameSite);
         }
 
-        //$response = $response->withHeader('Set-Cookie', implode(' ', $cookieValues));
-        header('Set-Cookie: ' . implode(' ', $cookieValues));
+        $response = $response->withHeader('Set-Cookie', implode(' ', $cookieValues));
 
         return $response;
     }
diff --git a/src/SameSiteSessionMiddleware.php b/src/SameSiteSessionMiddleware.php
index 6b82b53..f6340ce 100644
--- a/src/SameSiteSessionMiddleware.php
+++ b/src/SameSiteSessionMiddleware.php
@@ -20,11 +20,11 @@ final class SameSiteSessionMiddleware implements MiddlewareInterface
     /**
      * The constructor.
      *
-     * @param bool $startSession The the session
+     * @param SameSiteCookieConfiguration $configuration The configuration
      */
-    public function __construct(bool $startSession = true)
+    public function __construct(SameSiteCookieConfiguration $configuration)
     {
-        $this->startSession = $startSession;
+        $this->startSession = $configuration->startSession;
     }
 
     /**
diff --git a/tests/MiddlewareTestTrait.php b/tests/MiddlewareTestTrait.php
new file mode 100644
index 0000000..eb5d0fd
--- /dev/null
+++ b/tests/MiddlewareTestTrait.php
@@ -0,0 +1,41 @@
+<?php
+
+namespace Selective\SameSiteCookie\Test;
+
+use Nyholm\Psr7\Factory\Psr17Factory;
+use Psr\Http\Message\ResponseInterface;
+use Psr\Http\Message\ServerRequestInterface;
+use Relay\Relay;
+
+/**
+ * Test.
+ */
+trait MiddlewareTestTrait
+{
+    /**
+     * Run middleware stack.
+     *
+     * @param array $queue The queue
+     *
+     * @return ResponseInterface The response
+     */
+    protected function runQueue(array $queue): ResponseInterface
+    {
+        $queue[] = new ResponseFactoryMiddleware();
+
+        $request = $this->createRequest();
+        $relay = new Relay($queue);
+
+        return $relay->handle($request);
+    }
+
+    /**
+     * Factory.
+     *
+     * @return ServerRequestInterface
+     */
+    protected function createRequest(): ServerRequestInterface
+    {
+        return (new Psr17Factory())->createServerRequest('GET', '/');
+    }
+}
diff --git a/tests/ResponseFactoryMiddleware.php b/tests/ResponseFactoryMiddleware.php
new file mode 100644
index 0000000..5663f3b
--- /dev/null
+++ b/tests/ResponseFactoryMiddleware.php
@@ -0,0 +1,28 @@
+<?php
+
+namespace Selective\SameSiteCookie\Test;
+
+use Nyholm\Psr7\Factory\Psr17Factory;
+use Psr\Http\Message\ResponseInterface;
+use Psr\Http\Message\ServerRequestInterface;
+use Psr\Http\Server\MiddlewareInterface;
+use Psr\Http\Server\RequestHandlerInterface;
+
+/**
+ * Middleware.
+ */
+final class ResponseFactoryMiddleware implements MiddlewareInterface
+{
+    /**
+     * Invoke middleware.
+     *
+     * @param ServerRequestInterface $request The request
+     * @param RequestHandlerInterface $handler The handler
+     *
+     * @return ResponseInterface The response
+     */
+    public function process(ServerRequestInterface $request, RequestHandlerInterface $handler): ResponseInterface
+    {
+        return (new Psr17Factory())->createResponse();
+    }
+}
diff --git a/tests/SameSiteCookieMiddlewareTest.php b/tests/SameSiteCookieMiddlewareTest.php
index a511a73..cff3650 100644
--- a/tests/SameSiteCookieMiddlewareTest.php
+++ b/tests/SameSiteCookieMiddlewareTest.php
@@ -3,19 +3,35 @@
 namespace Selective\SameSiteCookie\Test;
 
 use PHPUnit\Framework\TestCase;
+use Selective\SameSiteCookie\SameSiteCookieConfiguration;
+use Selective\SameSiteCookie\SameSiteCookieMiddleware;
+use Selective\SameSiteCookie\SameSiteSessionMiddleware;
 
 /**
  * Test.
  */
 class SameSiteCookieMiddlewareTest extends TestCase
 {
+    use MiddlewareTestTrait;
+
     /**
      * Test.
      *
      * @return void
      */
-    public function testTrue(): void
+    public function testDefaultConfiguration(): void
     {
-        static::assertTrue(true);
+        $configuration = new SameSiteCookieConfiguration();
+
+        session_id('v3absd19o9pi6cjvhb5pkmsfo9');
+
+        $response = $this->runQueue([
+            new SameSiteSessionMiddleware($configuration),
+            new SameSiteCookieMiddleware($configuration),
+        ]);
+
+        $cookie = $response->getHeaderLine('Set-Cookie');
+        static::assertSame('PHPSESSID=v3absd19o9pi6cjvhb5pkmsfo9; path=/; Secure; HttpOnly; SameSite=Lax;', $cookie);
+        static::assertSame('', (string)$response->getBody());
     }
 }
diff --git a/tests/boostrap.php b/tests/boostrap.php
new file mode 100644
index 0000000..b8efaca
--- /dev/null
+++ b/tests/boostrap.php
@@ -0,0 +1,5 @@
+<?php
+
+require __DIR__ . '/../vendor/autoload.php';
+
+ob_start();