added some testing stuff in main