Added solution to week2, task3
Improved code for week2:task3