minor changed in main class