Add few docstrings & comments across the code