adding print to main function