[#26] Removes comments and fixes flake8 errors