fixed something in main