fixed main.m and added header