Add more integration test
Add rule permitting returning primitives/collections/strings in wrapper classes