Index: src/main/java/mk/ukim/finki/easyfood/service/impl/UserServiceImpl.java
===================================================================
--- src/main/java/mk/ukim/finki/easyfood/service/impl/UserServiceImpl.java	(revision 54cbfc74aca32192e82edf7cd78cba6d958669b3)
+++ src/main/java/mk/ukim/finki/easyfood/service/impl/UserServiceImpl.java	(revision 52af00c340f80b264edff12391e360f518423180)
@@ -86,5 +86,5 @@
         Customer customer = new Customer(email, passwordEncoder.encode(password), firstName, lastName, phoneNumber, ROLE.CUSTOMER);
 
-        return userRepository.save(customer);
+        return customerRepository.save(customer);
     }
 
Index: src/main/resources/templates/checkout.html
===================================================================
--- src/main/resources/templates/checkout.html	(revision 54cbfc74aca32192e82edf7cd78cba6d958669b3)
+++ src/main/resources/templates/checkout.html	(revision 52af00c340f80b264edff12391e360f518423180)
@@ -76,6 +76,7 @@
         <div class="text-center mb-4">
             <h1 class="fw-bold"><i class="fas fa-shopping-cart me-3"></i>Checkout</h1>
-            <p class="text-muted">Complete your order from <strong th:text="${restaurant.name}"></strong></p>
-        </div>
+            <p class="text-muted">Complete your order
+                <span th:if="${restaurant != null}">from <strong th:text="${restaurant.getName()}"></strong></span>
+            </p>        </div>
 
         <div class="row">
Index: src/main/resources/templates/profile.html
===================================================================
--- src/main/resources/templates/profile.html	(revision 54cbfc74aca32192e82edf7cd78cba6d958669b3)
+++ src/main/resources/templates/profile.html	(revision 52af00c340f80b264edff12391e360f518423180)
@@ -94,5 +94,5 @@
             <a href="/cart" class="btn btn-outline-secondary rounded-circle p-2 position-relative">
                 <img src="https://cdn-icons-png.flaticon.com/512/1170/1170678.png" alt="Cart" width="22" height="22">
-                <span class="position-absolute top-0 start-100 translate-middle badge rounded-pill bg-warning text-dark">2</span>
+                <span class="position-absolute top-0 start-100 translate-middle badge rounded-pill bg-warning text-dark">0</span>
             </a>
             <a href="/profile" class="btn btn-outline-secondary rounded-circle p-2">
@@ -319,4 +319,5 @@
     <script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0-alpha1/dist/js/bootstrap.bundle.min.js"
             crossorigin="anonymous"></script>
+</div>
 </body>
 </html>
Index: src/main/resources/templates/register_customer.html
===================================================================
--- src/main/resources/templates/register_customer.html	(revision 54cbfc74aca32192e82edf7cd78cba6d958669b3)
+++ src/main/resources/templates/register_customer.html	(revision 52af00c340f80b264edff12391e360f518423180)
@@ -14,5 +14,5 @@
         <!-- Logo -->
         <a class="navbar-brand d-flex align-items-center" href="index.html">
-            <img src="media/easyfood.png" alt="Logo" height="40">
+            <img src="/images/logo.JPG" alt="Logo" height="40">
         </a>
 
@@ -47,27 +47,36 @@
                         <!-- Form -->
                         <form class="form-signin mt-xl-5" method="post" action="/register">
+                            <!-- Add CSRF token -->
+                            <input type="hidden" th:name="${_csrf.parameterName}" th:value="${_csrf.token}"/>
+
                             <div class="mb-3">
                                 <label class="form-label fw-semibold">Full Name</label>
-                                <input type="text" name="fullName" class="form-control" value="">
+                                <input type="text" name="fullName" class="form-control" value="" required>
                             </div>
 
                             <div class="mb-3">
                                 <label class="form-label fw-semibold">Email</label>
-                                <input type="email" name="email" class="form-control" value="">
+                                <input type="email" name="email" class="form-control" value="" required>
                             </div>
 
                             <div class="mb-3">
                                 <label class="form-label fw-semibold">Phone Number</label>
-                                <input type="tel" name="phoneNumber" class="form-control" value="">
+                                <input type="tel" name="phoneNumber" class="form-control" value="" required>
                             </div>
 
                             <div class="mb-3">
                                 <label class="form-label fw-semibold">Password</label>
-                                <input type="password" name="password" class="form-control" placeholder="">
+                                <input type="password" name="password" class="form-control" placeholder="" required>
                             </div>
 
                             <div class="mb-3">
                                 <label class="form-label fw-semibold">Repeat Password</label>
-                                <input type="password" name="repeatedPassword" class="form-control" placeholder="">
+                                <input type="password" name="repeatedPassword" class="form-control" placeholder=""
+                                       required>
+                            </div>
+
+                            <!-- Display error message if any -->
+                            <div th:if="${error}" class="alert alert-danger" role="alert">
+                                <span th:text="${error}"></span>
                             </div>