Almost finished  AAE642 problem set 1