Move UI initialization to separate module