fixing up assignment to match solution better.