Changes to main class

hey i've added a new comment please update