Added task 2 in tdt4145