added test for additional labels
addressed review comments.