updated ui slightly.