Added some unit testing to ensure it is doing what I think it should be doing
The new files for the unit tests