Edits to correct data output