UI - styling fixes