Expanded the functionality of the program