minor ui clean up