Fix code examples in documentation

README.md

@@ -62,21 +62,28 @@ but for a taste of how the library works try the following code:
 
 ``` haskell
 import System.Console.ANSI
+import System.IO (stdout)
 
+main :: IO ()
 main = do
-    setCursorPosition 5 0
-    setTitle "ANSI Terminal Short Example"
-
-    setSGR [ SetConsoleIntensity BoldIntensity
-           , SetColor Foreground Vivid Red
-           ]
-    putStr "Hello"
-
-    setSGR [ SetConsoleIntensity NormalIntensity
-           , SetColor Foreground Vivid White
-           , SetColor Background Dull Blue
-           ]
-    putStrLn "World!"
+  stdoutSupportsANSI <- hNowSupportsANSI stdout
+  if stdoutSupportsANSI
+    then do
+      setCursorPosition 5 0
+      setTitle "ANSI Terminal Short Example"
+
+      setSGR [ SetConsoleIntensity BoldIntensity
+             , SetColor Foreground Vivid Red
+             ]
+      putStr "Hello"
+
+      setSGR [ SetConsoleIntensity NormalIntensity
+             , SetColor Foreground Vivid White
+             , SetColor Background Dull Blue
+             ]
+      putStrLn "World!"
+    else
+      putStrLn "Standard output does not support 'ANSI' escape codes."
 ```
 
 ![](https://raw.githubusercontent.com/feuerbach/ansi-terminal/master/example.png)

ansi-terminal/README.md

@@ -64,10 +64,11 @@ but for a taste of how the library works try the following code:
 import System.Console.ANSI
 import System.IO (stdout)
 
+main :: IO ()
 main = do
   stdoutSupportsANSI <- hNowSupportsANSI stdout
   if stdoutSupportsANSI
-    then
+    then do
       setCursorPosition 5 0
       setTitle "ANSI Terminal Short Example"
 

ansi-terminal/src/System/Console/ANSI.hs

@@ -138,7 +138,7 @@ A simple example is below:
 > main = do
 >   stdoutSupportsANSI <- hNowSupportsANSI stdout
 >   if stdoutSupportsANSI
->     then
+>     then do
 >       setSGR [SetColor Foreground Vivid Red]
 >       setSGR [SetColor Background Vivid Blue]
 >       putStrLn "Red-On-Blue"
@@ -158,7 +158,7 @@ Another example is below:
 > main = do
 >   stdoutSupportsANSI <- hNowSupportsANSI stdout
 >   if stdoutSupportsANSI
->     then
+>     then do
 >       setSGR [SetColor Foreground Dull Blue]
 >       putStr "Enter your name: "
 >       setSGR [SetColor Foreground Dull Yellow]