slight change with the return line being added. was looping