Add Test & Remove Unused Functions