Work on the TextBox
Still some bugs, but basically works