diff --git a/som.cabal b/som.cabal
--- a/som.cabal
+++ b/som.cabal
@@ -1,5 +1,5 @@
 name:           som
-version:        3.0
+version:        3.1
 synopsis:       Self-Organising Maps
 description:    A Kohonen Self-organising Map (SOM) maps input patterns 
                 onto a regular grid (usually two-dimensional) where each
diff --git a/src/Data/Datamining/Clustering/SOM.hs b/src/Data/Datamining/Clustering/SOM.hs
--- a/src/Data/Datamining/Clustering/SOM.hs
+++ b/src/Data/Datamining/Clustering/SOM.hs
@@ -23,13 +23,17 @@
 -- * Kohonen, T. (1982). Self-organized formation of topologically 
 --   correct feature maps. Biological Cybernetics, 43 (1), 59–69.
 --
+-- NOTE: Version 3.0 changed the order of parameters for many functions.
+-- This makes it easier for the user to write mapping and folding
+-- operations.
+--
 ------------------------------------------------------------------------
 
 {-# LANGUAGE UnicodeSyntax #-}
 
 module Data.Datamining.Clustering.SOM
   (
-    -- Patterns
+    -- * Patterns
     Pattern(..),
     -- * Using the SOM
     train,
diff --git a/src/Data/Datamining/Clustering/SOMInternal.hs b/src/Data/Datamining/Clustering/SOMInternal.hs
--- a/src/Data/Datamining/Clustering/SOMInternal.hs
+++ b/src/Data/Datamining/Clustering/SOMInternal.hs
@@ -10,10 +10,6 @@
 -- A module containing private @SOM@ internals. Most developers should
 -- use @SOM@ instead. This module is subject to change without notice.
 --
--- NOTE: Version 3.0 changed the order of parameters for many functions.
--- This makes it easier for the user to write mapping and folding
--- operations.
---
 ------------------------------------------------------------------------
 {-# LANGUAGE UnicodeSyntax, TypeFamilies, FlexibleContexts #-}
 
@@ -40,9 +36,8 @@
 import Data.Eq.Unicode ((≡))
 import Data.List (foldl', minimumBy)
 import Data.Ord (comparing)
-import Math.Geometry.GridMap (GridMap, Index, BaseGrid, distance, 
-  mapWithKey, toList)
-import Math.Geometry.Grid (Grid)
+import Math.Geometry.GridMap (GridMap, BaseGrid, mapWithKey, toList)
+import Math.Geometry.Grid (Grid, Index, distance)
 import qualified Math.Geometry.GridMap as GM (map)
 
 -- | A pattern to be learned or classified by a self-organising map.