Changed UI a little.