Do some design work to the UI