Bugfix. Removed logs
Improved UI