Added ability to write intergration tests