Updated ChatInputScreen