Cleaned the main.m