Add some small debug code to main method