Branch ui improvement