Small UI updates

Just fixing shit