V0.3.8: [updated] Map<TaskType, List<Task>> to map Task individually and test case
V0.3.8: [updated] fixed code to cater to new data type Map<TaskType, List<Task>>