Some small changes/improvements to the TextInput