Adding hot reloading