Second commit for AAE730 problem set 1