added more UI little things