Updated Main

Added main function to lab2 w/ comments