Working on the main() function