more work on numpy exercises