modify to allow for labels