First commit for hw2
Make test be more practical