Excercise 4 in TDT4190