Added a bunch of testing functionality to the program