Updating code to accept new format of content