Added second part of exercise - unfinished