Fixing textfield to work better