add some more touches for UI