Added integration

Removed some unnecessary code.