diff --git a/build.sbt b/build.sbt
index 3a6af78..4cc0fe4 100644
--- a/build.sbt
+++ b/build.sbt
@@ -5,6 +5,7 @@ name := "slick-codegen-example"
 inThisBuild(
   List(
     scalaVersion                        := "2.13.15",
+    scalacOptions += "-Xsource:3",
     githubWorkflowPublishTargetBranches := Seq(),
     githubWorkflowBuild += WorkflowStep.Sbt(List("run")),
     githubWorkflowJavaVersions          := Seq(JavaSpec.temurin("11"))
diff --git a/src/main/scala/Example.scala b/src/main/scala/slick/examples/codegen/Example.scala
similarity index 78%
rename from src/main/scala/Example.scala
rename to src/main/scala/slick/examples/codegen/Example.scala
index cc08c53..d10379a 100644
--- a/src/main/scala/Example.scala
+++ b/src/main/scala/slick/examples/codegen/Example.scala
@@ -1,15 +1,19 @@
+package slick.examples.codegen
+
+import slick.jdbc.H2Profile
+
 object Tables extends demo.Tables {
   // or just use object demo.Tables, which is hard-wired to the driver stated during generation
-  override val profile = slick.jdbc.H2Profile
+  override val profile: H2Profile.type = slick.jdbc.H2Profile
 }
 
 import scala.concurrent.Await
-import scala.concurrent.duration._
+import scala.concurrent.ExecutionContext.Implicits.global
+import scala.concurrent.duration.*
 import scala.language.postfixOps
 
-import Tables._
-import Tables.profile.api._
-import scala.concurrent.ExecutionContext.Implicits.global
+import slick.examples.codegen.Tables.*
+import slick.examples.codegen.Tables.profile.api.*
 
 object Example extends App {
   // connection info for a pre-populated throw-away, in-memory db for this demo, which is freshly initialized on every