Change output to stick with original exercise