Corrected minor ui bugs