Updated comments for class: Main