Updated comments for better readability