made UI prettier
fixed bug