DRY'd integration method
correction in integrate method
removed unused var
update
add_remove test