Adding main(), error free