updated main class of the project