add fucking tests(((