Added a feature to remove the keyboard from the screen when return is pressed