fix some bug in main class