Added tests to bring coverage back to 100%, fixed bug with labels.