Updated main.m