added func to get user input
Merge pull request #18 from yanhick/user-input

added func to get user input