Add few tests