writing integration tests
touch up