modification of the main class