Initiated task 3, chapter 7
Added argument parsing