diff --git a/docs/stories/index.js b/docs/stories/index.js
index e39f4d0..9010830 100644
--- a/docs/stories/index.js
+++ b/docs/stories/index.js
@@ -7,6 +7,6 @@ import { storiesOf } from '@storybook/react';
 
 storiesOf('HTML source', module).add('basic', html.basic);
 
-storiesOf('URI source', module).add('basic', uri.basic).add('onMessage', uri.onMessage);
+storiesOf('URI source', module).add('basic', uri.basic).add('post message', uri.onMessage);
 
 storiesOf('With method', module).add('basic', method.basic);
diff --git a/docs/stories/method.js b/docs/stories/method.js
index 09ed1a6..a4624e5 100644
--- a/docs/stories/method.js
+++ b/docs/stories/method.js
@@ -3,10 +3,12 @@ import { WebView } from 'react-native-webview';
 import { text } from '@storybook/addon-knobs';
 
 export const basic = () => (
-  <WebView
-    source={{
-      uri: text('URI', 'https://react-native-web-community.github.io/react-native-web-webview/'),
-      method: text('Method', 'GET'),
-    }}
-  />
+  <div style={{ height: '100vh' }}>
+    <WebView
+      source={{
+        uri: text('URI', 'https://react-native-web-community.github.io/react-native-web-webview/'),
+        method: text('Method', 'GET'),
+      }}
+    />
+  </div>
 );
diff --git a/docs/stories/onMessage.html b/docs/stories/onMessage.html
index e48f079..452f18e 100644
--- a/docs/stories/onMessage.html
+++ b/docs/stories/onMessage.html
@@ -1,6 +1,8 @@
 <html>
   <body>
     <h1>Hello world!</h1>
+
+    <input type="text" id="message-input"  placeholder="enter your message">
     <button id="button">Send message</button>
     <script type="text/javascript">
       // React Native
@@ -14,11 +16,10 @@ <h1>Hello world!</h1>
         }
       }
 
-      var payload = JSON.stringify({
-        name: 'WebView',
-      });
+
 
       document.getElementById('button').addEventListener('click', function () {
+        let payload = document.getElementById("message-input").value || '';
         if (window.opener) {
           // Web new window
           window.opener.postMessage(payload, window.opener.origin);
diff --git a/docs/stories/uri.js b/docs/stories/uri.js
index 98de3dc..c834f43 100644
--- a/docs/stories/uri.js
+++ b/docs/stories/uri.js
@@ -1,10 +1,20 @@
 import React from 'react';
 import { WebView } from 'react-native-webview';
 import { action } from '@storybook/addon-actions';
-import { text } from '@storybook/addon-knobs';
 
-export const basic = () => <WebView source={{ uri: 'https://www.youtube.com/embed/dQw4w9WgXcQ' }} />;
+export const basic = () => {
+  return (
+    <div style={{ height: '100vh' }}>
+      <WebView source={{ uri: 'https://www.youtube.com/embed/dQw4w9WgXcQ' }} />
+    </div>
+  );
+};
 
-export const onMessage = () => (
-  <WebView source={{ uri: require('./onMessage.html') }} onMessage={action(text('Text', 'onMessage'))} />
-);
+export const onMessage = () => {
+  return (
+    <WebView
+      source={{ uri: require('./onMessage.html') }}
+      onMessage={(e) => action('message from webview')(e.nativeEvent.data)}
+    />
+  );
+};
