Added initialization for task 2 and first tries implementing it