I have added some features to the program.