Added enter key functionality