Adding comments and fixing pep8