More token styling & input height fixes