Updated main.m.
More edits.