Functions WS 20 - added user input