Fixed setting task as done, Added rest of unit tests