update the UI to match WP better