Add main tests back
Try adding print statements