Start of task 7

Just added the function