Updating to match the proper return dict and proper casing