diff --git a/README.md b/README.md
index 2dab0d6..4669e19 100644
--- a/README.md
+++ b/README.md
@@ -11,8 +11,9 @@ app_file: space.py
 
 <h1 style='text-align: center; margin-bottom: 1rem'> Gradio WebRTC ⚡️ </h1>
 
-<div>
-<img style="display: block; margin-left: auto; margin-right: auto" alt="Static Badge" src="https://img.shields.io/badge/version%20-%200.0.5%20-%20orange">  
+<div style="display: flex; flex-direction: row; justify-content: center">
+<img style="display: block; padding-right: 5px; height: 20px;" alt="Static Badge" src="https://img.shields.io/badge/version%20-%200.0.5%20-%20orange"> 
+<a href="https://github.com/freddyaboulton/gradio-webrtc" target="_blank"><img alt="Static Badge" src="https://img.shields.io/badge/github-white?logo=github&logoColor=black"></a>
 </div>
 
 <h3 style='text-align: center'>
diff --git a/demo/space.py b/demo/space.py
index ef039ab..a0a85f2 100644
--- a/demo/space.py
+++ b/demo/space.py
@@ -39,13 +39,10 @@ with gr.Blocks(
 """
 <h1 style='text-align: center; margin-bottom: 1rem'> Gradio WebRTC ⚡️ </h1>
 
-<div>
-<img style="display: block; margin-left: auto; margin-right: auto" alt="Static Badge" src="https://img.shields.io/badge/version%20-%200.0.4%20-%20orange">  
+<div style="display: flex; flex-direction: row; justify-content: center">
+<img style="display: block; padding-right: 5px; height: 20px;" alt="Static Badge" src="https://img.shields.io/badge/version%20-%200.0.5%20-%20orange"> 
+<a href="https://github.com/freddyaboulton/gradio-webrtc" target="_blank"><img alt="Static Badge" src="https://img.shields.io/badge/github-white?logo=github&logoColor=black"></a>
 </div>
-
-<h3 style='text-align: center'>
-Stream video and audio in real time with Gradio using WebRTC. 
-</h3>
 """, elem_classes=["md-custom"], header_links=True)
     gr.Markdown(
 """