Made some fine changes to UI