finished numpy ex 1