Added partial solution for challenge 3