Minor change to problem #7