#7 added in user input implementation