Added on_enter function for TextField