diff --git a/frontend/src/preferences/PromptsModal.tsx b/frontend/src/preferences/PromptsModal.tsx
index 721564e..da95dc0 100644
--- a/frontend/src/preferences/PromptsModal.tsx
+++ b/frontend/src/preferences/PromptsModal.tsx
@@ -60,7 +60,7 @@ function AddPrompt({
         <div className="mt-6 flex flex-row-reverse">
           <button
             onClick={handleAddPrompt}
-            className="rounded-3xl bg-purple-30 px-6 py-3 text-sm text-white transition-all hover:opacity-90"
+            className="rounded-3xl bg-purple-30 px-5 py-2 text-sm text-white transition-all hover:opacity-90"
           >
             Save
           </button>
@@ -130,7 +130,7 @@ function EditPrompt({
         </div>
         <div className="mt-6 flex flex-row-reverse gap-4">
           <button
-            className={`rounded-3xl bg-purple-30 px-6 py-3 text-sm text-white transition-all ${
+            className={`rounded-3xl bg-purple-30 px-5 py-2 text-sm text-white transition-all ${
               currentPromptEdit.type === 'public'
                 ? 'cursor-not-allowed opacity-50'
                 : 'hover:opacity-90'
diff --git a/frontend/src/upload/Upload.tsx b/frontend/src/upload/Upload.tsx
index 066d674..4abf263 100644
--- a/frontend/src/upload/Upload.tsx
+++ b/frontend/src/upload/Upload.tsx
@@ -308,7 +308,7 @@ export default function Upload({
                 setUrlType(value)
               }
               size="w-full"
-              rounded="xl"
+              rounded="3xl"
             />
             {urlType.label !== 'Reddit' ? (
               <>
@@ -419,7 +419,7 @@ export default function Upload({
             disabled={
               (files.length === 0 || docName.trim().length === 0) &&
               activeTab === 'file'
-            } // Disable the button if no file is selected or docName is empty
+            }
           >
             Train
           </button>
@@ -450,4 +450,3 @@ export default function Upload({
     </article>
   );
 }
-// TODO: sanitize all inputs