Fixes UI details

Fixes UI details