Sample code for stage_8_1
Possible solution for the exercise