refactor Control class methods to take predict function as argument