Solution to the third challenge - level 2