diff --git a/mermaid/__main__.py b/mermaid/__main__.py
index cceed5b..35d480a 100644
--- a/mermaid/__main__.py
+++ b/mermaid/__main__.py
@@ -1,177 +1,178 @@
-import base64
-from enum import Enum
-from pathlib import Path
-from typing import Optional, Union
-from urllib.parse import urlencode
-
-import requests
-from requests import Response
-
-from .graph import Graph
-
-
-class Position(Enum):
-    """
-    This class represents the position of the node in a Mermaid diagram.
-    """
-
-    LEFT = "left"
-    RIGHT = "right"
-    CENTER = "center"
-    NONE = "none"
-
-
-class Mermaid:
-    """
-    This class represents a Mermaid diagram.
-
-    Attributes:
-        _diagram (str): The base64 encoded string of the Mermaid diagram script.
-        svg_response (Response): The response from the GET request to the Mermaid SVG API.
-        img_response (Response): The response from the GET request to the Mermaid IMG API.
-    """
-
-    def __init__(
-        self,
-        graph: Graph,
-        width: Optional[int] = None,
-        height: Optional[int] = None,
-        scale: Optional[float] = None,
-        position: Union[Position, str] = Position.NONE,
-    ):
-        """
-        The constructor for the Mermaid class.
-
-        Parameters:
-            graph (Graph): The Graph object containing the Mermaid diagram script.
-            width (Optional[int]): The width of the SVG image.
-            height (Optional[int]): The height of the SVG image.
-            scale (Optional[float]): The scale of the SVG image.
-                Must be an float between 1 and 3, and one of height or width must be provided.
-            position (Union[Position, str]): The position of the node in the Mermaid diagram.
-        """
-        if scale:
-            assert 1 <= scale <= 3, "Scale must be between 1 and 3"
-            assert any(
-                [width, height]
-            ), "One or both of width and height must be provided"
-
-        self.__position: str = position if isinstance(position, str) else position.value
-        self.__height = height if height else None
-        self.__width = width if width else None
-        self.__scale = scale if scale else None
-
-        self._diagram = self._process_diagram(graph.script)
-
-        if any([self.__width, self.__height, self.__scale]):
-            self._diagram += "?" + self._build_query_params()
-        self._make_request_to_mermaid()
-
-    def _build_query_params(self) -> str:
-        """Build the query parameters for the Mermaid API request."""
-        params = {
-            param: value
-            for param, value in [
-                ("width", self.__width),
-                ("height", self.__height),
-                ("scale", self.__scale),
-            ]
-            if value
-        }
-
-        return urlencode(params, doseq=True)
-
-    def set_position(self, position: Union[Position, str]) -> None:
-        """
-        Set the position of the node in the Mermaid diagram.
-
-        Parameters:
-            position (Union[Position, str]): The position of the node.
-        """
-        self.__position = position if isinstance(position, str) else position.value
-
-    @staticmethod
-    def _process_diagram(diagram: str) -> str:
-        """
-        Process the Mermaid diagram script into a base64 encoded string.
-
-        Parameters:
-            diagram (str): The Mermaid diagram script.
-
-        Returns:
-            str: The base64 encoded string of the Mermaid diagram script.
-        """
-        graphbytes = diagram.encode("utf8")
-        base64_bytes = base64.b64encode(graphbytes)
-        diagram = base64_bytes.decode("ascii")
-        return diagram
-
-    def _repr_html_(self) -> str:
-        """
-        Return the text of the SVG response.
-
-        Returns:
-            str: The text of the SVG response.
-        """
-        if self.__position == Position.NONE.value:
-            return self.svg_response.text
-        return (
-            f'<div style="text-align:{self.__position}">{self.svg_response.text}</div>'
-        )
-
-    def _make_request_to_mermaid(self) -> None:
-        """
-        Make GET requests to the Mermaid SVG and IMG APIs using
-        the base64 encoded string of the Mermaid diagram script.
-        """
-
-        self.svg_response: Response = requests.get(
-            "https://mermaid.ink/svg/" + self._diagram
-        )
-        self.img_response: Response = requests.get(
-            "https://mermaid.ink/img/" + self._diagram
-        )
-
-    def to_svg(self, path: Union[str, Path]) -> None:
-        """
-        Write the SVG response text to a file.
-
-        Parameters:
-            path (Union[str, Path]): The path of the file to write to.
-        """
-        with open(path, "w", encoding="utf-8") as file:
-            file.write(self.svg_response.text)
-
-    def to_png(self, path: Union[str, Path]) -> None:
-        """
-        Write the IMG response content to a file.
-
-        Parameters:
-            path (Union[str, Path]): The path of the file to write to.
-        """
-        with open(path, "wb") as file:
-            file.write(self.img_response.content)
-
-
-try:
-    from IPython import get_ipython
-
-    if get_ipython() is not None:
-        from IPython.core.magic import register_cell_magic
-        from IPython.display import Image, display
-
-        @register_cell_magic
-        def mermaidjs(line, cell):
-            options = line.strip().split()
-            script: str = cell.strip()
-            graph: Graph = Graph("mermaid diagram", script)
-            mermaid: Mermaid = Mermaid(graph)
-            if "--img" in options:
-                display(Image(mermaid.img_response.content))
-            else:
-                display(Mermaid(graph))
-
-        get_ipython().register_magic_function(mermaidjs, magic_kind="cell")
-except ImportError:
-    # TODO: add a suitible handler for exception.
-    print("Error acured while importing mermaidjs .")
+import base64
+from enum import Enum
+from pathlib import Path
+from typing import Optional, Union
+from urllib.parse import urlencode
+
+import requests
+from requests import Response
+
+from .graph import Graph
+
+
+class Position(Enum):
+    """
+    This class represents the position of the node in a Mermaid diagram.
+    """
+
+    LEFT = "left"
+    RIGHT = "right"
+    CENTER = "center"
+    NONE = "none"
+
+
+class Mermaid:
+    """
+    This class represents a Mermaid diagram.
+
+    Attributes:
+        _diagram (str): The base64 encoded string of the Mermaid diagram script.
+        svg_response (Response): The response from the GET request to the Mermaid SVG API.
+        img_response (Response): The response from the GET request to the Mermaid IMG API.
+    """
+
+    def __init__(
+        self,
+        graph: Graph,
+        width: Optional[int] = None,
+        height: Optional[int] = None,
+        scale: Optional[float] = None,
+        position: Union[Position, str] = Position.NONE,
+    ):
+        """
+        The constructor for the Mermaid class.
+
+        Parameters:
+            graph (Graph): The Graph object containing the Mermaid diagram script.
+            width (Optional[int]): The width of the SVG image.
+            height (Optional[int]): The height of the SVG image.
+            scale (Optional[float]): The scale of the SVG image.
+                Must be an float between 1 and 3, and one of height or width must be provided.
+            position (Union[Position, str]): The position of the node in the Mermaid diagram.
+        """
+        if scale:
+            assert 1 <= scale <= 3, "Scale must be between 1 and 3"
+            assert any(
+                [width, height]
+            ), "One or both of width and height must be provided"
+
+        self.__position: str = position if isinstance(position, str) else position.value
+        self.__height = height if height else None
+        self.__width = width if width else None
+        self.__scale = scale if scale else None
+
+        self._diagram = self._process_diagram(graph.script)
+
+        if any([self.__width, self.__height, self.__scale]):
+            self._diagram += "?" + self._build_query_params()
+        self._make_request_to_mermaid()
+
+    def _build_query_params(self) -> str:
+        """Build the query parameters for the Mermaid API request."""
+        params = {
+            param: value
+            for param, value in [
+                ("width", self.__width),
+                ("height", self.__height),
+                ("scale", self.__scale),
+            ]
+            if value
+        }
+
+        return urlencode(params, doseq=True)
+
+    def set_position(self, position: Union[Position, str]) -> None:
+        """
+        Set the position of the node in the Mermaid diagram.
+
+        Parameters:
+            position (Union[Position, str]): The position of the node.
+        """
+        self.__position = position if isinstance(position, str) else position.value
+
+    @staticmethod
+    def _process_diagram(diagram: str) -> str:
+        """
+        Process the Mermaid diagram script into a base64 encoded string.
+
+        Parameters:
+            diagram (str): The Mermaid diagram script.
+
+        Returns:
+            str: The base64 encoded string of the Mermaid diagram script.
+        """
+        graphbytes = diagram.encode("utf8")
+        base64_bytes = base64.b64encode(graphbytes)
+        diagram = base64_bytes.decode("ascii")
+        return diagram
+
+    def _repr_html_(self) -> str:
+        """
+        Return the text of the SVG response.
+
+        Returns:
+            str: The text of the SVG response.
+        """
+        if self.__position == Position.NONE.value:
+            return self.svg_response.text
+        return (
+            f'<div style="text-align:{self.__position}">{self.svg_response.text}</div>'
+        )
+
+    def _make_request_to_mermaid(self) -> None:
+        """
+        Make GET requests to the Mermaid SVG and IMG APIs using
+        the base64 encoded string of the Mermaid diagram script.
+        """
+
+        self.svg_response: Response = requests.get(
+            "https://mermaid.ink/svg/" + self._diagram
+        )
+        self.img_response: Response = requests.get(
+            "https://mermaid.ink/img/" + self._diagram
+        )
+
+    def to_svg(self, path: Union[str, Path]) -> None:
+        """
+        Write the SVG response text to a file.
+
+        Parameters:
+            path (Union[str, Path]): The path of the file to write to.
+        """
+        with open(path, "w", encoding="utf-8") as file:
+            file.write(self.svg_response.text)
+
+    def to_png(self, path: Union[str, Path]) -> None:
+        """
+        Write the IMG response content to a file.
+
+        Parameters:
+            path (Union[str, Path]): The path of the file to write to.
+        """
+        with open(path, "wb") as file:
+            file.write(self.img_response.content)
+
+
+try:
+    from IPython import get_ipython
+
+    if get_ipython() is not None:
+        from IPython.core.magic import register_cell_magic
+        from IPython.display import Image, display
+
+        @register_cell_magic
+        def mermaidjs(line, cell):
+            options = line.strip().split()
+            script: str = cell.strip()
+            graph: Graph = Graph("mermaid diagram", script)
+            mermaid: Mermaid = Mermaid(graph)
+            if "--img" in options:
+                display(Image(mermaid.img_response.content))
+            else:
+                display(Mermaid(graph))
+
+        get_ipython().register_magic_function(mermaidjs, magic_kind="cell")
+except ImportError:
+    print(
+        "Warning: IPython is not installed. Mermaidjs magic function is not available."
+    )