use hyper::header::{ContentLength, Headers};
use std::ops::Deref;

use Bytes;

/// Trait to extend functionalities of the Headers type, from `hyper`
pub trait GetContentLength {
    /// Function to get the content length of a remote document.
    /// The returned type is `Option<Bytes>`.
    fn get_content_length(&self) -> Option<Bytes>;
}

impl GetContentLength for Headers {
    /// Function to get the `content-length` container, from a given header.
    /// This function returns an Option that contains a `Bytes` type.
    fn get_content_length(&self) -> Option<Bytes> {
        if self.has::<ContentLength>() {
            return Some(*self.get::<ContentLength>().unwrap().deref());
        }
        None
    }
}

use hyper::header::{ContentLength, Headers};
use std::ops::Deref;

use Byte;

/// Trait to extend functionalities of the Headers type, from `hyper`
pub trait GetContentLength {
    /// Function to get the content length of a remote document.
    /// The returned type is `Byte`.
    fn get_content_length(&self) -> Option<Byte>;
}

impl GetContentLength for Headers {
    /// Function to get the `content-length` container, from a given header.
    /// This function returns an Option that contains a `Byte` type.
    fn get_content_length(&self) -> Option<Byte> {
        if self.has::<ContentLength>() {
            return Some(*self.get::<ContentLength>().unwrap().deref())
        }
        None
    }
}
--- 
+++ 
@@ -1,21 +1,21 @@
 use hyper::header::{ContentLength, Headers};
 use std::ops::Deref;
 
-use Byte;
+use Bytes;
 
 /// Trait to extend functionalities of the Headers type, from `hyper`
 pub trait GetContentLength {
     /// Function to get the content length of a remote document.
-    /// The returned type is `Byte`.
-    fn get_content_length(&self) -> Option<Byte>;
+    /// The returned type is `Option<Bytes>`.
+    fn get_content_length(&self) -> Option<Bytes>;
 }
 
 impl GetContentLength for Headers {
     /// Function to get the `content-length` container, from a given header.
-    /// This function returns an Option that contains a `Byte` type.
-    fn get_content_length(&self) -> Option<Byte> {
+    /// This function returns an Option that contains a `Bytes` type.
+    fn get_content_length(&self) -> Option<Bytes> {
         if self.has::<ContentLength>() {
-            return Some(*self.get::<ContentLength>().unwrap().deref())
+            return Some(*self.get::<ContentLength>().unwrap().deref());
         }
         None
     }
Update Byte type to Bytes
