small change to make function to get return data