Updates: Almost finish the task4 part c)