Update the rake file to include step output