Add enter handling in input