change return type back to Array