Fixed fixed better UI

It's better now, Kevin