Add main function
Add output of tests