Almost done with the exercise