Output cleanup for beginner solution