Adding comments for code readability