Add some testing stuff