Completed first assignment with code and plots