fixed main.m