Adding some needed funtionality for the exercise to function properly.