tweaks to output. All tests now passing! :)