BUG: Forgot to add the file.. Added doctest as well.