Added complete test main method