Cleaned up the ui
Never forget to update the readme!