Fixed two of the errors showing in main.m