Add test.

Add few basic test.