Added comments for code documentation