Changed a little UI