Added a few unit test. Need add alot more.