Messed around with the UI to fix it up.