Add tests for updateStyle and fix bug when it should remove all styles