Setup unit test

I'm so sorry that I forgot to implement unit test!