added integration tests + class diagrams