diff --git a/docs/css/main.css b/docs/css/main.css
index 99edeba..22955cf 100755
--- a/docs/css/main.css
+++ b/docs/css/main.css
@@ -202,6 +202,10 @@ iframe {
 	justify-content: center;
 }
 
+.header a.leave {
+	margin-left: auto;
+}
+
 .clickie {
 	width: min-content;
 	margin-top: 40px;
diff --git a/docs/index.php b/docs/index.php
index 2c12284..5c7caff 100644
--- a/docs/index.php
+++ b/docs/index.php
@@ -7,7 +7,7 @@ page_header();
 if (!is_null($username)) {
     //comments("root","root");
 }
-if (!($_GET['page'] > 0)) { ?>	
+if (!($_GET['page'] > 0) && is_null($username)) { ?>	
 		<div class="banner"><img class="banner-background" src="/img/newlogo.svg">
 			<div class="banner-content">
 				<h1>BiglyChat</h1>
diff --git a/docs/register.php b/docs/register.php
index d99c4ed..004c402 100755
--- a/docs/register.php
+++ b/docs/register.php
@@ -44,6 +44,12 @@ function post_handler()
     $stmt = $db->prepare("INSERT INTO main.auth (username, password) VALUES (?, ?)");
     $stmt->execute([$user,$hashed]);
 
+ 	$token = bin2hex(random_bytes(32));
+    $stmt = $db->prepare("INSERT INTO main.token (username, token) VALUES (?, ?)");
+    $stmt->execute([$user,$token]);
+
+    setcookie("token", $token, time() + 3600 * 24);
+
     header("Location: /");
     die();
 }
diff --git a/libs/page.php b/libs/page.php
index cfd9c0a..c250979 100755
--- a/libs/page.php
+++ b/libs/page.php
@@ -29,7 +29,7 @@ function page_header()
 						<a href="/user.php?id=<?php echo $username ?>"><img class="header-img header-link" src="/pfp/<?php echo $username ?>.png">Wall</a>
 								<a href="/notif.php"> <img class="header-img header-link" src="/img/mail.svg"><span>Mail<sub> <?php echo $edge?></sub></span></a>
 								<a href="/stats.php?username=<?php echo $username ?>"><img class="header-img header-link" src="/img/home.svg">Users</a>
-								<a href="/logout.php"><img class="header-img header-link" src="/img/logout.svg">Leave</a>
+								<a href="/logout.php" class='leave'><img class="header-img header-link" src="/img/logout.svg">Leave</a>
 								<?php } else { ?>
 										<a href="/login.php"><img class="header-img header-link" src="/img/home.svg">Join</a>
 												<a href="/stats.php"><img class="header-img header-link" src="/img/home.svg">Users</a>