More updates to ui styling