add solution to the third assignement