Add estimation functionality