completed numpy exercise