Improved the UI
Removed print statements