Optimize UI
Fix bug