Adding couple integration tests