Added some unit tests to project