Add user input function