add test
working now