diff --git a/README.md b/README.md
--- a/README.md
+++ b/README.md
@@ -1,5 +1,7 @@
 # plot-gtk-ui
 
+[![Build Status](https://travis-ci.org/sumitsahrawat/plot-gtk-ui.svg)](https://travis-ci.org/sumitsahrawat/plot-gtk-ui)
+
 An ambitious attempt to provide mathematica like dynamic plotting for free.
 
 ## Installation
diff --git a/lib/Graphics/Rendering/Plot/Gtk/UI.hs b/lib/Graphics/Rendering/Plot/Gtk/UI.hs
--- a/lib/Graphics/Rendering/Plot/Gtk/UI.hs
+++ b/lib/Graphics/Rendering/Plot/Gtk/UI.hs
@@ -98,8 +98,8 @@
               -> [(Double, Double)]   -- ^ Ranges for all args
               -> Maybe (IO ())
 plotWithArity n func ranges =
-    if length ranges /= n
-    then Nothing
-    else Just $ plotUnsafe func ranges
+  if length ranges /= n
+  then Nothing
+  else Just $ plotUnsafe func ranges
 
 --------------------------------------------------------------------------------
diff --git a/lib/Graphics/Rendering/Plot/Gtk/UI/Figure.hs b/lib/Graphics/Rendering/Plot/Gtk/UI/Figure.hs
--- a/lib/Graphics/Rendering/Plot/Gtk/UI/Figure.hs
+++ b/lib/Graphics/Rendering/Plot/Gtk/UI/Figure.hs
@@ -28,7 +28,7 @@
 import           Graphics.Rendering.Plot
 import           Graphics.UI.Gtk                         (Adjustment,
                                                           adjustmentGetValue)
-import           Numeric.LinearAlgebra                   (linspace, mapVector)
+import           Numeric.LinearAlgebra                   (linspace, cmap)
 
 --------------------------------------------------------------------------------
 -- Custom Modules
@@ -77,7 +77,7 @@
       domain  = linspace samples range
       func    = g vars
       stype   = plotType fset
-      dset    = [(stype, domain, mapVector func domain)]
+      dset    = [(stype, domain, cmap func domain)]
   return $ buildFigure dset fset
 
 --------------------------------------------------------------------------------
diff --git a/plot-gtk-ui.cabal b/plot-gtk-ui.cabal
--- a/plot-gtk-ui.cabal
+++ b/plot-gtk-ui.cabal
@@ -6,7 +6,7 @@
 -- PVP summary:      +-+------- breaking API changes
 --                   | | +----- non-breaking API additions
 --                   | | | +--- code changes with no API change
-Version:             0.3.0.0
+Version:             0.3.0.1
 License:             GPL-2
 License-file:        LICENSE
 Author:              Sumit Sahrawat
@@ -23,7 +23,7 @@
 
 Category:            Math
 
-Tested-with:         GHC==7.8.3
+Tested-with:         GHC==7.8.3, GHC==7.8.4, GHC==7.10.1, GHC==7.10.2
 Cabal-version:       >=1.18
 Build-type:          Simple
 
@@ -32,7 +32,7 @@
 Data-files:          plot-window.glade
 
 Library
-  Build-depends:       base >=4.7 && <4.9
+  Build-depends:       base >=4.6 && <4.9
                      , colour
                      , plot
                      , gtk
diff --git a/stack.yaml b/stack.yaml
--- a/stack.yaml
+++ b/stack.yaml
@@ -1,7 +1,7 @@
-# For more information, see: https://github.com/commercialhaskell/stack/blob/master/doc/yaml_configuration.md
+# For more information, see: https://github.com/commercialhaskell/stack/blob/release/doc/yaml_configuration.md
 
 # Specifies the GHC version and set of packages available (e.g., lts-3.5, nightly-2015-09-21, ghc-7.10.2)
-resolver: lts-3.6
+resolver: lts-3.8
 
 # Local packages, usually specified by relative directory name
 packages:
@@ -12,6 +12,9 @@
 
 # Override default flag values for local packages and extra-deps
 flags: {}
+
+# Extra package databases containing global packages
+extra-package-dbs: []
 
 # Control whether we use the GHC we find on the path
 # system-ghc: true