add integration tests
check todo