add assignment 1: answers for numpy