Nearly finished the exercise