Add two unit test