added main to the program