Fixing build and displaying UI