refining numpy ex 1