Added initial implementation of task 2 from week 4.