Added better comments for the main class