[~FEATURE] Added integration tests