diff --git a/demo/jest.config.ts b/demo/jest.config.ts
index 2bb9f9d9..0d4c754f 100644
--- a/demo/jest.config.ts
+++ b/demo/jest.config.ts
@@ -19,7 +19,7 @@ const config: Config = {
     clearMocks: true,
 
     // Indicates whether the coverage information should be collected while executing the test
-    collectCoverage: true,
+    collectCoverage: false,
 
     // An array of glob patterns indicating a set of files for which coverage information should be collected
     // collectCoverageFrom: undefined,
@@ -46,6 +46,7 @@ const config: Config = {
     // An object that configures minimum threshold enforcement for coverage results
     // coverageThreshold: undefined,
 
+    displayName: "jsdom tests for demo website",
     // A path to a custom dependency extractor
     // dependencyExtractor: undefined,
 
diff --git a/demo/package.json b/demo/package.json
index 5c52fc37..1b378d40 100644
--- a/demo/package.json
+++ b/demo/package.json
@@ -3,7 +3,8 @@
     "version": "0.1.0",
     "description": "Demo website for memory-models-rough",
     "scripts": {
-        "test": "jest --coverage",
+        "test": "jest --no-cache",
+        "test-cov": "jest --no-cache --coverage",
         "build-dev": "webpack",
         "start": "webpack serve"
     },
diff --git a/demo/src/SvgDisplay.tsx b/demo/src/SvgDisplay.tsx
index 9c02f38b..dd4d10d0 100644
--- a/demo/src/SvgDisplay.tsx
+++ b/demo/src/SvgDisplay.tsx
@@ -1,5 +1,5 @@
 import React, { useRef, useEffect } from "react";
-import mem from "../../src/index";
+import mem from "../../src/index"; // TODO: replace with import of the package after it's been published
 
 export default function SvgDisplay(props) {
     const canvasRef = useRef(null);
diff --git a/demo/src/__tests__/SvgDisplay.spec.tsx b/demo/src/__tests__/SvgDisplay.spec.tsx
index 0e0779e8..cfd649b9 100644
--- a/demo/src/__tests__/SvgDisplay.spec.tsx
+++ b/demo/src/__tests__/SvgDisplay.spec.tsx
@@ -34,7 +34,6 @@ describe("SvgDisplay", () => {
             ];
             render(<SvgDisplay jsonResult={jsonResult} />);
         });
-        afterEach(cleanup);
 
         it("renders canvas element with specified dimensions", () => {
             const canvasElement = screen.getByTestId("memory-models-canvas");
diff --git a/jest.config.ts b/jest.config.ts
index 7408adea..02ce481b 100644
--- a/jest.config.ts
+++ b/jest.config.ts
@@ -16,10 +16,10 @@ const config: Config = {
     // cacheDirectory: "/private/var/folders/tl/mq0ry9mn37n2_djks873jbj80000gn/T/jest_dx",
 
     // Automatically clear mock calls, instances, contexts and results before every test
-    clearMocks: true,
+    // clearMocks: true,
 
     // Indicates whether the coverage information should be collected while executing the test
-    collectCoverage: true,
+    // collectCoverage: false,
 
     // An array of glob patterns indicating a set of files for which coverage information should be collected
     // collectCoverageFrom: undefined,
@@ -90,10 +90,7 @@ const config: Config = {
     // ],
 
     // A map from regular expressions to module names or to arrays of module names that allow to stub out resources with a single module
-    moduleNameMapper: {
-        // Force module roughjs to resolve with the CJS entry point, because Jest does not support package.json.exports. Elaborated in PR#15.
-        roughjs: require.resolve("roughjs"),
-    },
+    // moduleNameMapper: {},
 
     // An array of regexp pattern strings, matched against all module paths before considered 'visible' to the module loader
     // modulePathIgnorePatterns: [],
@@ -108,7 +105,7 @@ const config: Config = {
     // preset: undefined,
 
     // Run tests from one or more projects
-    // projects: undefined,
+    projects: ["./demo/jest.config.ts", "./src/jest.config.ts"],
 
     // Use this configuration option to add custom reporters to Jest
     // reporters: undefined,
@@ -149,7 +146,7 @@ const config: Config = {
     // snapshotSerializers: [],
 
     // The test environment that will be used for testing
-    testEnvironment: "node",
+    // testEnvironment: "node",
 
     // Options that will be passed to the testEnvironment
     // testEnvironmentOptions: {},
diff --git a/package.json b/package.json
index 2dcca24d..06e645bd 100644
--- a/package.json
+++ b/package.json
@@ -5,7 +5,7 @@
     "main": "dist/memory_models_rough.js",
     "scripts": {
         "test": "jest --no-cache",
-        "test-cov": "jest --no-cache --collect-coverage",
+        "test-cov": "jest --no-cache --coverage",
         "watch": "webpack --watch",
         "build-dev": "tsc && webpack",
         "build": "tsc && webpack --mode production",
diff --git a/src/jest.config.ts b/src/jest.config.ts
new file mode 100644
index 00000000..98b0a1b3
--- /dev/null
+++ b/src/jest.config.ts
@@ -0,0 +1,203 @@
+/**
+ * For a detailed explanation regarding each configuration property, visit:
+ * https://jestjs.io/docs/configuration
+ */
+
+import type { Config } from "jest";
+
+const config: Config = {
+    // All imported modules in your tests should be mocked automatically
+    // automock: false,
+
+    // Stop running tests after `n` failures
+    // bail: 0,
+
+    // The directory where Jest should store its cached dependency information
+    // cacheDirectory: "/private/var/folders/tl/mq0ry9mn37n2_djks873jbj80000gn/T/jest_dx",
+
+    // Automatically clear mock calls, instances, contexts and results before every test
+    clearMocks: true,
+
+    // Indicates whether the coverage information should be collected while executing the test
+    collectCoverage: false,
+
+    // An array of glob patterns indicating a set of files for which coverage information should be collected
+    // collectCoverageFrom: undefined,
+
+    // The directory where Jest should output its coverage files
+    coverageDirectory: "coverage",
+
+    // An array of regexp pattern strings used to skip coverage collection
+    // coveragePathIgnorePatterns: [
+    //   "/node_modules/"
+    // ],
+
+    // Indicates which provider should be used to instrument code for coverage
+    // coverageProvider: "babel",
+
+    // A list of reporter names that Jest uses when writing coverage reports
+    // coverageReporters: [
+    //   "json",
+    //   "text",
+    //   "lcov",
+    //   "clover"
+    // ],
+
+    // An object that configures minimum threshold enforcement for coverage results
+    // coverageThreshold: undefined,
+
+    displayName: "node tests for memory models scripts",
+    // A path to a custom dependency extractor
+    // dependencyExtractor: undefined,
+
+    // Make calling deprecated APIs throw helpful error messages
+    // errorOnDeprecated: false,
+
+    // The default configuration for fake timers
+    // fakeTimers: {
+    //   "enableGlobally": false
+    // },
+
+    // Force coverage collection from ignored files using an array of glob patterns
+    // forceCoverageMatch: [],
+
+    // A path to a module which exports an async function that is triggered once before all test suites
+    // globalSetup: undefined,
+
+    // A path to a module which exports an async function that is triggered once after all test suites
+    // globalTeardown: undefined,
+
+    // A set of global variables that need to be available in all test environments
+    // globals: {},
+
+    // The maximum amount of workers used to run your tests. Can be specified as % or a number. E.g. maxWorkers: 10% will use 10% of your CPU amount + 1 as the maximum worker number. maxWorkers: 2 will use a maximum of 2 workers.
+    // maxWorkers: "50%",
+
+    // An array of directory names to be searched recursively up from the requiring module's location
+    // moduleDirectories: [
+    //   "node_modules"
+    // ],
+
+    // An array of file extensions your modules use
+    // moduleFileExtensions: [
+    //   "js",
+    //   "mjs",
+    //   "cjs",
+    //   "jsx",
+    //   "ts",
+    //   "tsx",
+    //   "json",
+    //   "node"
+    // ],
+
+    // A map from regular expressions to module names or to arrays of module names that allow to stub out resources with a single module
+    moduleNameMapper: {
+        // Force module roughjs to resolve with the CJS entry point, because Jest does not support package.json.exports. Elaborated in PR#15.
+        roughjs: require.resolve("roughjs"),
+    },
+
+    // An array of regexp pattern strings, matched against all module paths before considered 'visible' to the module loader
+    // modulePathIgnorePatterns: [],
+
+    // Activates notifications for test results
+    // notify: false,
+
+    // An enum that specifies notification mode. Requires { notify: true }
+    // notifyMode: "failure-change",
+
+    // A preset that is used as a base for Jest's configuration
+    // preset: undefined,
+
+    // Run tests from one or more projects
+    // projects: undefined,
+
+    // Use this configuration option to add custom reporters to Jest
+    // reporters: undefined,
+
+    // Automatically reset mock state before every test
+    // resetMocks: false,
+
+    // Reset the module registry before running each individual test
+    // resetModules: false,
+
+    // A path to a custom resolver
+    // resolver: undefined,
+
+    // Automatically restore mock state and implementation before every test
+    // restoreMocks: false,
+
+    // The root directory that Jest should scan for tests and modules within
+    // rootDir: undefined,
+
+    // A list of paths to directories that Jest should use to search for files in
+    // roots: [
+    //   "<rootDir>"
+    // ],
+
+    // Allows you to use a custom runner instead of Jest's default test runner
+    // runner: "jest-runner",
+
+    // The paths to modules that run some code to configure or set up the testing environment before each test
+    // setupFiles: [],
+
+    // A list of paths to modules that run some code to configure or set up the testing framework before each test
+    // setupFilesAfterEnv: [],
+
+    // The number of seconds after which a test is considered as slow and reported as such in the results.
+    // slowTestThreshold: 5,
+
+    // A list of paths to snapshot serializer modules Jest should use for snapshot testing
+    // snapshotSerializers: [],
+
+    // The test environment that will be used for testing
+    testEnvironment: "node",
+
+    // Options that will be passed to the testEnvironment
+    // testEnvironmentOptions: {},
+
+    // Adds a location field to test results
+    // testLocationInResults: false,
+
+    // The glob patterns Jest uses to detect test files
+    // testMatch: [
+    //   "**/__tests__/**/*.[jt]s?(x)",
+    //   "**/?(*.)+(spec|test).[tj]s?(x)"
+    // ],
+
+    // An array of regexp pattern strings that are matched against all test paths, matched tests are skipped
+    // testPathIgnorePatterns: [
+    //   "/node_modules/"
+    // ],
+
+    // The regexp pattern or array of patterns that Jest uses to detect test files
+    // testRegex: [],
+
+    // This option allows the use of a custom results processor
+    // testResultsProcessor: undefined,
+
+    // This option allows use of a custom test runner
+    // testRunner: "jest-circus/runner",
+
+    // A map from regular expressions to paths to transformers
+    // transform: undefined,
+
+    // An array of regexp pattern strings that are matched against all source file paths, matched files will skip transformation
+    // transformIgnorePatterns: [
+    //   "/node_modules/",
+    //   "\\.pnp\\.[^\\/]+$"
+    // ],
+
+    // An array of regexp pattern strings that are matched against all modules before the module loader will automatically return a mock for them
+    // unmockedModulePathPatterns: undefined,
+
+    // Indicates whether each individual test should be reported during the run
+    // verbose: undefined,
+
+    // An array of regexp patterns that are matched against all source file paths before re-running tests in watch mode
+    // watchPathIgnorePatterns: [],
+
+    // Whether to use watchman for file crawling
+    // watchman: true,
+};
+
+export default config;
diff --git a/tests/__snapshots__/draw.spec.tsx.snap b/src/tests/__snapshots__/draw.spec.tsx.snap
similarity index 100%
rename from tests/__snapshots__/draw.spec.tsx.snap
rename to src/tests/__snapshots__/draw.spec.tsx.snap
diff --git a/tests/draw.spec.tsx b/src/tests/draw.spec.tsx
similarity index 96%
rename from tests/draw.spec.tsx
rename to src/tests/draw.spec.tsx
index a31bff8f..3acc508c 100644
--- a/tests/draw.spec.tsx
+++ b/src/tests/draw.spec.tsx
@@ -1,4 +1,4 @@
-import exports from "../src";
+import exports from "../index";
 const { MemoryModel, draw } = exports;
 
 describe("draw function", () => {