Add Unit Test

Include some unit tests for this project