Added more functionality to the program.