First attempt of adding solution for Prog Ass 2