Tweaked UI slightly.