Added new function to comply with part 1 question 5