changed a bit of the UI