need to add user input