Merge Task 1 and Task 2 
Add input funtion to  Task 2