Added nicer UI. fixes #115