diff --git a/Dockerfile b/Dockerfile
index c096a80..0a17daa 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,22 +1,23 @@
-FROM node:20-alpine
+FROM node:18-alpine
 
-# Create app directory
 WORKDIR /app
 
-# Install app dependencies
+# Install dependencies
 COPY package*.json ./
 RUN npm install
 
-# Bundle app source
+# Copy app source
 COPY . .
 
-# Create uploads directory
-RUN mkdir -p public/uploads
+# Create necessary directories with correct permissions
+RUN mkdir -p /app/data /app/public/uploads && \
+    chown -R node:node /app/data /app/public/uploads
 
-# Create non-root user
-RUN addgroup -S appgroup && adduser -S appuser -G appgroup
-RUN chown -R appuser:appgroup /app
-USER appuser
+# Switch to non-root user
+USER node
 
-EXPOSE 3000
-CMD [ "node", "server.js" ]
\ No newline at end of file
+# Expose port
+EXPOSE 2222
+
+# Start the app
+CMD ["node", "server.js"]
\ No newline at end of file
diff --git a/config/database.js b/config/database.js
index 99c7112..932f414 100644
--- a/config/database.js
+++ b/config/database.js
@@ -2,106 +2,114 @@ const sqlite3 = require('sqlite3').verbose();
 const path = require('path');
 const bcrypt = require('bcrypt');
 const winston = require('winston');
+const fs = require('fs');
+
+// Ensure data directory exists
+const dataDir = path.resolve(__dirname, '../data');
+if (!fs.existsSync(dataDir)) {
+    fs.mkdirSync(dataDir, { recursive: true });
+}
 
 const logger = winston.createLogger({
-  level: 'info',
-  format: winston.format.combine(
-    winston.format.timestamp(),
-    winston.format.json()
-  ),
-  transports: [
-    new winston.transports.File({ filename: 'data/app.log' }),
-    new winston.transports.Console()
-  ]
+    level: 'info',
+    format: winston.format.combine(
+        winston.format.timestamp(),
+        winston.format.json()
+    ),
+    transports: [
+        new winston.transports.File({ filename: path.join(dataDir, 'app.log') }),
+        new winston.transports.Console()
+    ]
 });
 
-const dbPath = path.resolve(__dirname, '../data/notes.db');
+const dbPath = path.join(dataDir, 'notes.db');
+logger.info(`Using database at: ${dbPath}`);
 
 const db = new sqlite3.Database(dbPath, (err) => {
-  if (err) {
-    logger.error('Database connection error:', err);
-    process.exit(1);
-  }
-  logger.info('Connected to SQLite database');
-  initializeDatabase();
+    if (err) {
+        logger.error('Database connection error:', err);
+        process.exit(1);
+    }
+    logger.info('Connected to SQLite database');
+    initializeDatabase();
 });
 
 function initializeDatabase() {
-  db.serialize(() => {
-    // Users table
-    db.run(`CREATE TABLE IF NOT EXISTS users (
-      id INTEGER PRIMARY KEY AUTOINCREMENT,
-      username TEXT UNIQUE NOT NULL,
-      password_hash TEXT NOT NULL,
-      is_admin INTEGER DEFAULT 0,
-      created_at DATETIME DEFAULT CURRENT_TIMESTAMP
-    )`);
+    db.serialize(() => {
+        // Users table
+        db.run(`CREATE TABLE IF NOT EXISTS users (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            username TEXT UNIQUE NOT NULL,
+            password_hash TEXT NOT NULL,
+            is_admin INTEGER DEFAULT 0,
+            created_at DATETIME DEFAULT CURRENT_TIMESTAMP
+        )`);
 
-    // Notes table
-    db.run(`CREATE TABLE IF NOT EXISTS notes (
-      id INTEGER PRIMARY KEY AUTOINCREMENT,
-      user_id INTEGER NOT NULL,
-      title TEXT NOT NULL,
-      content TEXT,
-      archived INTEGER DEFAULT 0,
-      created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-      updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
-      FOREIGN KEY (user_id) REFERENCES users(id)
-    )`);
+        // Notes table
+        db.run(`CREATE TABLE IF NOT EXISTS notes (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            user_id INTEGER NOT NULL,
+            title TEXT NOT NULL,
+            content TEXT,
+            archived INTEGER DEFAULT 0,
+            created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY (user_id) REFERENCES users(id)
+        )`);
 
-    // Todos table
-    db.run(`CREATE TABLE IF NOT EXISTS todos (
-      id INTEGER PRIMARY KEY AUTOINCREMENT,
-      note_id INTEGER NOT NULL,
-      text TEXT NOT NULL,
-      completed INTEGER DEFAULT 0,
-      position INTEGER NOT NULL,
-      FOREIGN KEY (note_id) REFERENCES notes(id)
-    )`);
+        // Todos table
+        db.run(`CREATE TABLE IF NOT EXISTS todos (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            note_id INTEGER NOT NULL,
+            text TEXT NOT NULL,
+            completed INTEGER DEFAULT 0,
+            position INTEGER NOT NULL,
+            FOREIGN KEY (note_id) REFERENCES notes(id)
+        )`);
 
-    // Images table
-    db.run(`CREATE TABLE IF NOT EXISTS images (
-      id INTEGER PRIMARY KEY AUTOINCREMENT,
-      note_id INTEGER NOT NULL,
-      filename TEXT NOT NULL,
-      upload_date DATETIME DEFAULT CURRENT_TIMESTAMP,
-      FOREIGN KEY (note_id) REFERENCES notes(id)
-    )`);
+        // Images table
+        db.run(`CREATE TABLE IF NOT EXISTS images (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            note_id INTEGER NOT NULL,
+            filename TEXT NOT NULL,
+            upload_date DATETIME DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY (note_id) REFERENCES notes(id)
+        )`);
 
-    // Create indexes
-    db.run('CREATE INDEX IF NOT EXISTS idx_notes_user_id ON notes(user_id)');
-    db.run('CREATE INDEX IF NOT EXISTS idx_notes_created_at ON notes(created_at)');
-    db.run('CREATE INDEX IF NOT EXISTS idx_todos_note_id ON todos(note_id)');
-    db.run('CREATE INDEX IF NOT EXISTS idx_images_note_id ON images(note_id)');
+        // Create indexes
+        db.run('CREATE INDEX IF NOT EXISTS idx_notes_user_id ON notes(user_id)');
+        db.run('CREATE INDEX IF NOT EXISTS idx_notes_created_at ON notes(created_at)');
+        db.run('CREATE INDEX IF NOT EXISTS idx_todos_note_id ON todos(note_id)');
+        db.run('CREATE INDEX IF NOT EXISTS idx_images_note_id ON images(note_id)');
 
-    // Check if admin user exists
-    db.get('SELECT * FROM users WHERE username = ?', ['admin'], (err, row) => {
-      if (err) {
-        logger.error('Error checking admin user:', err);
-        return;
-      }
-      if (!row) {
-        // Create default admin user
-        bcrypt.hash('admin', 10, (err, hash) => {
-          if (err) {
-            logger.error('Error hashing password:', err);
-            return;
-          }
-          db.run(
-            'INSERT INTO users (username, password_hash, is_admin) VALUES (?, ?, 1)',
-            ['admin', hash],
-            (err) => {
-              if (err) {
-                logger.error('Error creating admin user:', err);
+        // Check if admin user exists
+        db.get('SELECT * FROM users WHERE username = ?', ['admin'], (err, row) => {
+            if (err) {
+                logger.error('Error checking admin user:', err);
                 return;
-              }
-              logger.info('Default admin user created');
             }
-          );
+            if (!row) {
+                // Create default admin user
+                bcrypt.hash('admin', 10, (err, hash) => {
+                    if (err) {
+                        logger.error('Error hashing password:', err);
+                        return;
+                    }
+                    db.run(
+                        'INSERT INTO users (username, password_hash, is_admin) VALUES (?, ?, 1)',
+                        ['admin', hash],
+                        (err) => {
+                            if (err) {
+                                logger.error('Error creating admin user:', err);
+                                return;
+                            }
+                            logger.info('Default admin user created');
+                        }
+                    );
+                });
+            }
         });
-      }
     });
-  });
 }
 
 module.exports = { db, logger };
\ No newline at end of file