Add TEST + todo function