Assignment changes to add 2 functions.