Applied pep 8 and updated comments.