Minor pep 8 and comment changes