WIP: add integration testing