diff --git a/connect.js b/lib/index.js
similarity index 91%
rename from connect.js
rename to lib/index.js
index bf83d79..eb8667b 100644
--- a/connect.js
+++ b/lib/index.js
@@ -3,12 +3,12 @@ const path = require('path');
 const glob = require('glob');
 const rimraf = require('rimraf');
 const mkdirp = require('mkdirp');
-const create_routes = require('./lib/utils/create_routes.js');
-const templates = require('./lib/templates.js');
-const create_app = require('./lib/utils/create_app.js');
-const create_compiler = require('./lib/utils/create_compiler.js');
+const create_routes = require('./utils/create_routes.js');
+const templates = require('./templates.js');
+const create_app = require('./utils/create_app.js');
+const create_compiler = require('./utils/create_compiler.js');
 const escape_html = require('escape-html');
-const { src, dest, dev } = require('./lib/config.js');
+const { src, dest, dev } = require('./config.js');
 
 module.exports = function connect(opts) {
 	mkdirp(dest);
diff --git a/lib/utils/create_compiler.js b/lib/utils/create_compiler.js
index 3870adb..0957055 100644
--- a/lib/utils/create_compiler.js
+++ b/lib/utils/create_compiler.js
@@ -2,6 +2,7 @@ const fs = require('fs');
 const path = require('path');
 const glob = require('glob');
 const webpack = require('webpack');
+const hot_middleware = require('webpack-hot-middleware');
 const { dev } = require('../config.js');
 const templates = require('../templates.js');
 
@@ -77,6 +78,8 @@ module.exports = function create_webpack_compiler(dest, routes, dev) {
 		}
 
 		if (dev) {
+			compiler.hot_middleware = hot_middleware(client);
+
 			let client_is_ready = false;
 			let server_is_ready = false;