Added in two unit test