Solution for the second chall