Working with functionality for enter key