update main.m