Added task headings and test cases