Add sum fn & test