Improving the testing module and removing some prints