added two unit tests