Refactoring itm to sklearn class api, remove duplication from experiments