editing a bit in main function