Adding test for main