From 446c83a6cf0c323ca0d39a16ad0f3a3bfe851c29 Mon Sep 17 00:00:00 2001
From: Yuan Zhao <yuanzhaoYZ@users.noreply.github.com>
Date: Wed, 13 Mar 2019 10:51:03 -0500
Subject: [PATCH] Update README.md (#48)

---
 README.md | 32 ++++++++++++++++----------------
 1 file changed, 16 insertions(+), 16 deletions(-)

diff --git a/README.md b/README.md
index 45970f6..0b8b198 100644
--- a/README.md
+++ b/README.md
@@ -117,34 +117,34 @@ trait SparkSessionTestWrapper {
 The `DatasetComparer` trait defines the `assertSmallDatasetEquality` method.  Extend your spec file with the `SparkSessionTestWrapper` trait to create DataFrames and the `DatasetComparer` trait to make DataFrame comparisons.
 
 ```scala
+import org.apache.spark.sql.types._
+import org.apache.spark.sql.Row
+import org.apache.spark.sql.functions._
 import com.github.mrpowers.spark.fast.tests.DatasetComparer
 
 class DatasetSpec extends FunSpec with SparkSessionTestWrapper with DatasetComparer {
 
   import spark.implicits._
 
-    it("aliases a DataFrame") {
+  it("aliases a DataFrame") {
 
-      val sourceDF = Seq(
-        ("jose"),
-        ("li"),
-        ("luisa")
-      ).toDF("name")
+    val sourceDF = Seq(
+      ("jose"),
+      ("li"),
+      ("luisa")
+    ).toDF("name")
 
-      val actualDF = sourceDF.select(col("name").alias("student"))
+    val actualDF = sourceDF.select(col("name").alias("student"))
 
-      val expectedDF = Seq(
-        ("jose"),
-        ("li"),
-        ("luisa")
-      ).toDF("student")
+    val expectedDF = Seq(
+      ("jose"),
+      ("li"),
+      ("luisa")
+    ).toDF("student")
 
-      assertSmallDatasetEquality(actualDF, expectedDF)
-
-    }
+    assertSmallDatasetEquality(actualDF, expectedDF)
 
   }
-
 }
 ```