progress on numpy exercise 1